import argparse

def parse_args():
    parser = argparse.ArgumentParser()
    parser.add_argument('--dataset',nargs='?',default="tiktok",help="数据集")
    parser.add_argument('--data_path',default='F:\\code\\python_code\\LightGCN-self\\data\\',help="数据集位置")
    parser.add_argument('--keep_rate',type=float,default=0.8)
    parser.add_argument('--emb_size',type=int,default=128)
    parser.add_argument('--gpu_id',type=int, default=0, help='GPU id')
    parser.add_argument('--lr',type=float,default=0.0006)
    parser.add_argument('--d_lr', type=float, default=0.006)
    parser.add_argument('--decay', type=float, default=1e-5)
    parser.add_argument('--batch_size',type=int,default=1024)
    parser.add_argument('--epoch',type=int,default=1000)
    parser.add_argument('--head_num', type=int, default=4)
    parser.add_argument('--Ks', nargs='?', default='[10, 20, 50]', help='K value of ndcg/recall @ k')
    parser.add_argument('--early_stop', type=int, default=7, help='')
    parser.add_argument('--is_vision', type=int, default=1, help='')
    parser.add_argument('--is_grad', type=int, default=0, help='')
    parser.add_argument('--hyper_graph_layers', type=int, default=1, help='')
    parser.add_argument('--nei_layers', type=int, default=2, help='')
    parser.add_argument('--n_layers', type=int, default=2, help='')
    parser.add_argument('--model_cat_rate',type=float,default=0.25)
    parser.add_argument('--hyper_rate',type=float,default=0.002,help='超图稀疏度')
    parser.add_argument('--hyper_num', type=int, default=10, help='超图节点最小值')
    parser.add_argument('--model', nargs='?', default="Light_GCN()", help="[Light_GCN(),Hyper_only(),MLP_Diss(),MLP_Att(),Gan_Att(),Sp_Hyper()]")
    return parser.parse_args()

args = parse_args()
